import requests
import app


# 封装登录类、方法
class LoginAPI(object):
    def __init__(self):
        self.url = app.BASE_URL+'/api/sys/login'
        self.headers = app.HEADERS

    def login(self, json_data):
        # 封装 http 请求 的 登录方法
        resp = requests.post(url=self.url, headers=self.headers, json=json_data)
        return resp


# 封装 员工管理类、方法
class EmpManage(object):
    def __init__(self):
        self.url = app.BASE_URL + '/api/sys/user'

    # 添加员工，获取id
    def add_emp(self, headers, req_data):
        return requests.post(url=self.url, headers=headers, json=req_data)

    # 查询员工，使用id
    def query_emp(self, emp_id, headers):
        return requests.get(url=self.url+'/'+emp_id, headers=headers)

    # 修改员工，使用id
    def modify_emp(self, emp_id, headers, req_data):
        return requests.put(url=self.url+'/'+emp_id, headers=headers, json=req_data)

    # 删除员工，使用id
    def delete_emp(self, emp_id, headers):
        return requests.delete(url=self.url+'/'+emp_id, headers=headers)

    # 查询员工列表
    def query_emp_list(self, headers):
        return requests.get(url=app.BASE_URL+'/api/sys/user?page=1&size=10', headers=headers)